"""
综合案例--南波万ATM
"""
money = 1000000
def mean():
    """
    打印ATM主界面，输入要办理的业务
    :return: 返回值是选择办理的业务的代码
    """
    print("-----------------主菜单------------------")
    print("学分，您好，欢迎来到南波万银行ATM，请您选择操作：")
    print("查询余额\t[输入1]")
    print("存款 \t[输入2]")
    print("取款 \t[输入3]")
    print("退出 \t[输入4]")
    return input("请输入您的选择：")

def query():
    """
    查询余额功能
    :return: None
    """
    print("----------查询余额-----------")
    print(f"学分，您好，您的余额剩余{money}")

def save(num):
    """
    实现存款功能，
    :param num:需要存的钱数
    :return: None
    """
    print("-------------存款--------------")
    print(f"学分，您好，您存款{num}元")
    global money
    money+=num
    print(f"学分，您好，您的余额剩余：{money}元")

def remove(num):
    """
        实现取款功能，
        :param num:需要取的钱数
        :return: None
        """
    print("-------------取款--------------")
    print(f"学分，您好，您取款{num}元")
    global money
    money -= num
    print(f"学分，您好，您的余额剩余：{money}元")

while True:
    my_input = mean()
    if my_input == "1":
        query()
    elif my_input == "2":
        num = int(input("您想要存多少钱，请输入："))
        save(num)
    elif my_input == "3":
        num = int(input("您想要取多少钱，请输入："))
        remove(num)
    elif my_input == "4":
        print("退出程序")
        break
    else:
        print("输入错误，请重新输入")